#include "base_util/utils.h"
#include "base_util/math_compute.h"
#include "stream/osd_text.h"
#include "common/face_tool.h"
#include "model_manager.h"


// 调用示例函数
void performance_test(ai::ModelManager* model, std::string file_path, std::vector<ai::InitModelData> init_model_datas, int max_channel_size);
int infer_det_img(ai::ModelManager* model, stream::CVOsdText* osd, std::vector<ai::InitModelData>& init_model_datas, std::string file_path);
int infer_det_img_multi(ai::ModelManager* model, stream::CVOsdText* osd, std::vector<ai::InitModelData>& init_model_datas, std::string file_path);
int infer_det_mp4(ai::ModelManager* model, stream::CVOsdText* osd, std::vector<ai::InitModelData>& init_model_datas, std::string mp4_path);
int infer_det_img_api(ai::ModelManager* model, std::vector<ai::InitModelData>& init_model_datas, std::string input_str, std::string& ret_str);
int infer_det_img_to_via_file(ai::ModelManager* model, stream::CVOsdText* osd, std::vector<ai::InitModelData>& init_model_datas, std::string file_path);
int infer_cls_video(ai::ModelManager* model, stream::CVOsdText* osd, std::vector<ai::InitModelData>& init_model_datas, std::string file_path);
int infer_face_img_det_rec(ai::ModelManager* model, stream::CVOsdText* osd, std::string file_path);
